import argparse

DEEPER_FORENSICS = 'deeperface'


def parse_args():
    parser = argparse.ArgumentParser(description="Parameters of preprocessing and statistical program")
    parser.add_argument("--root-dir", default="G:\\DeeperForensics-1.0", help="Root directory")
    parser.add_argument("--ori-dir", default="G:\\FaceForensics++", help="Root directory")
    parser.add_argument("--fake-type", type=str, default="end_to_end_level_1", help="Manipulation type")
    parser.add_argument("--detector-type", default="FacenetDetector", choices=["FacenetDetector"],
                        help="type of the detector")
    parser.add_argument("--crops-dir", default="end_to_end_level_1_crops", help="crops directory")
    parser.add_argument("--diffs-dir", default="end_to_end_level_1_diffs", help="diffs directory")
    args = parser.parse_args()
    return args
